from pydantic import BaseModel


class PredictRequest(BaseModel):
    """ 预测请求体类型 """
    text: str


class PredictResponse(BaseModel):
    """ 预测响应体类型 """
    text: str
    pred_id: int
    pred_label: str
